import torch
from model import DeepCADModel

# 初始化模型
model = DeepCADModel()

# 加载预训练权重
model.load_state_dict(torch.load('/data-extend/guoyimo/CAD/DeepCAD/pretrained/model/ckpt_epoch1000.pth'))

# 生成 CAD 模型
cad_model = model.generate()

# 保存生成的 CAD 模型
cad_model.save('output_cad_model.cad')